#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2021/7/23 6:41 上午
# @Author : lihc
# @File : home_page.py

# 每个黄色方块代表一个类
from selenium.webdriver.common.by import By

from seleniumdemo.po.add_member_page import AddMemberPage
from seleniumdemo.po.base_page import BasePage
from seleniumdemo.po.contact_page import ContactPage


class HomePage(BasePage):
    _menu_contacts = (By.ID, 'menu_contacts')
    _home_add_member = (By.CSS_SELECTOR, '.index_service_cnt_itemWrap:nth-child(1)')

    # 页面地址
    def __init__(self, base_driver=None):
        self.read_config_file()
        self._home_page_url = self.conn.get('URL', 'home_url')
        # 初始化Base init
        BasePage.__init__(self)
    
    # 点击通讯录Tab
    def click_contact(self):
         '''
         方法应该返回其他的PageObject或者返回用于断言的数据
         箭头的终点代表跳转页面
         跳转到通讯录页面
         :return:
         '''
         # 点击通讯录Tab

         self.find(*self._menu_contacts).click()

         # 跳转到通讯录页面
         return ContactPage()


    # 点击添加成员链接
    def click_add_member(self):
        '''
         跳转到添加成员页面
        :return:
        '''

        # 点击添加成员按钮
        self.find(*self._home_add_member).click()

        # 跳转到添加成员页面
        return AddMemberPage(self.driver)